Python program to create and write to a Text File